import os

# 基础路径配置
BASE_DIR = os.path.dirname(os.path.abspath(__file__))

# 模型配置
MODEL_CONFIG = {
    'path': 'checkpoints/checkpoint_resnet50.pth.tar',
    'input_size': 224,
    'mean': [0.485, 0.456, 0.406],
    'std': [0.229, 0.224, 0.225]
}

# 数据集配置
DATASET_CONFIG = {
    'path': os.path.join(BASE_DIR, '垃圾图片库'),
    'label_file': 'dir_label.txt'
}

# 缓存配置
CACHE_CONFIG = {
    'feature_dir': 'feature_cache',
    'max_age': 7 * 24 * 60 * 60,  # 7天的缓存过期时间
    'cleanup_interval': 24 * 60 * 60  # 每24小时清理一次过期缓存
}

# 上传配置
UPLOAD_CONFIG = {
    'allowed_extensions': {'png', 'jpg', 'jpeg', 'gif'},
    'max_content_length': 5 * 1024 * 1024  # 5MB
}

# 搜索配置
SEARCH_CONFIG = {
    'top_k': 5,
    'similarity_threshold': 0.5
}

# 日志配置
LOG_CONFIG = {
    'filename': 'app.log',
    'level': 'INFO',
    'format': '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
}
